package com.sxj.shop.servlet;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;

import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @author 石小俊
 * @date 2024年03月22日 18:02
 */
@WebServlet("*.file")
public class FileServlet extends HttpServlet {
    @Override
    protected void service(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        String path = request.getServletPath();
        if ("/upload.file".equals(path)) {
            upload(request, response);
        }
        if ("/download.file".equals(path)) {
            download(request, response);
        }
        if ("/download2.file".equals(path)) {
            download2(request, response);
        }
    }

    public void download(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        // 获取需要下载的资源地址
        String path = getServletContext().getRealPath("/js/jquery-3.4.1.min.js");

        FileInputStream in = null;
        OutputStream out = null;
        try {
            in = new FileInputStream(path);
            out = response.getOutputStream();

//            byte[] b = new byte[10240];
//            int i;
//            while ((i = in.read(b)) != -1){
//                out.write(b,0,i);
//            }

            // 此时执行后浏览器做的是在线打开操作
            // 可以手动定义下载方式
            // 通过content-disposition属性进行设置
            // 其值有两个:attachment(附件下载)、inline(在线打开)
//            response.setHeader("content-disposition","attachment");
            // 此时设置附件下载之后,下载出来的文件的名字是以当前请求命名的
            // 我们可以自定义文件名,同样还是content-disposition属性
            // 其值格式为:下载方式;filename=文件名
//            response.setHeader("content-disposition","attachment;filename=jquery-3.4.1.min.js");
            // 如果文件名中包含中文,需要将其转换为ISO-8859-1的字符集,否则中文无效
//            response.setHeader("content-disposition","attachment;filename=中文.min.js");
//            response.setHeader("content-disposition",new String("attachment;filename=中文.min.js".getBytes(),"ISO-8859-1"));
            response.setHeader("content-disposition", "attachment;filename=jquery-3.4.1.min.js");

            byte[] b = new byte[10240];
            int i;
            while ((i = in.read(b)) != -1) {
                out.write(b, 0, i);
            }

        } catch (FileNotFoundException e) {
            e.printStackTrace();
        }

    }


    public void download2(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        // 获取需要下载的资源地址
        String path = getServletContext().getRealPath("/lib/jstl.jar");

        FileInputStream in = null;
        OutputStream out = null;
        try {
            in = new FileInputStream(path);
            out = response.getOutputStream();

//            byte[] b = new byte[10240];
//            int i;
//            while ((i = in.read(b)) != -1){
//                out.write(b,0,i);
//            }

//            response.setHeader("content-disposition","attachment");
            response.setHeader("content-disposition", "attachment;filename=jstl.jar");
            byte[] b = new byte[10240];
            int i;
            while ((i = in.read(b)) != -1) {
                out.write(b, 0, i);
            }
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        }

    }


    public void upload(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        // 该方式只能获取表单数据
//        String username = request.getParameter("username");
//        String file = request.getParameter("file");
//        System.out.println("username:" + username + ",file:" + file);

        // 如果想要获取二进制流数据,需要手动进行解析
//        InputStream in = request.getInputStream();
//        byte[] b = new byte[10240];
//        int i = in.read(b);
//        System.out.println(new String(b,0,i));
        // 通过上述方式已经可以获取到请求中的二进制数据了
        // 但是由于数据较为混乱,需要对获取到的数据做解析
        // 这个解析操作是比较麻烦的,但是解析成功之后就能实现文件上传了
        // 由于这个解析操作比较麻烦,因此很多开发者就针对性的研发了对应的依赖
        // 流行最广的是Apache提供的commons-fileupload
        // 注意点:该依赖只是做了解析,并不是上传

        // 使用第三方工具进行解析
        // 获取解析工厂
        DiskFileItemFactory factory = new DiskFileItemFactory();
        // 获取解析器
        ServletFileUpload upload = new ServletFileUpload(factory);

        try {
            // 进行解析
            List<FileItem> fileItems = upload.parseRequest(request);
            Map<String, FileItem> items = new HashMap<>();
            for (FileItem fileItem : fileItems) {
//                System.out.println(fileItem.getFieldName());
//                if("username".equals(fileItem.getFieldName())){
//                    System.out.println("用户名:"+fileItem.getString());
//                }
//                if("file".equals(fileItem.getFieldName())){
//                    System.out.println("准备上传");
//                }
                items.put(fileItem.getFieldName(), fileItem);
            }
            System.out.println("用户名:" + items.get("username").getString("UTF-8"));

            // 准备上传
            // 上传到服务器中,在服务器根目录中创建一个目录upload,专门存放所有上传的文件
            // 在一个项目中,上传的文件可能会非常多,为了避免混乱,在upload中创建子目录
            // 每天创建一个以当前年月日命名的子目录,用于存放当天上传的文件
            // 想要获取服务器地址,可以通过application对象来获取
            ServletContext application = getServletContext();
            // getRealPath(path):获取服务器地址
            // 参数表示相对与服务器根目录的地址
            // 如果参数为空字符串,表示获取服务器根目录
            // 参数的作用是影响返回值的,其最终返回值为:服务器跟目录+path
            String path = application.getRealPath("/upload/" + new SimpleDateFormat("yyyyMMdd").format(new Date()));
            System.out.println(path);
            File f = new File(path);
            // 创建该文件所表示的路径中涉及的所有目录
            f.mkdirs();
            FileItem file = items.get("file");
            String fileName = file.getName();

            // 不允许上传jsp文件
            if (fileName.endsWith(".jsp")) {
                request.setAttribute("uploadMsg", "不允许上传JSP文件");
                request.getRequestDispatcher("/jsp/upload.jsp").forward(request, response);
                return;
            }

            fileName = fileName.substring(0, fileName.lastIndexOf(".")) + "-" + new Date().getTime() + fileName.substring(fileName.lastIndexOf("."));
            file.write(new File(path, fileName));
        } catch (FileUploadException e) {
            e.printStackTrace();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}
